#!/bin/bash

BACKEND_FILE="server.js"
PLACEHOLDER="{API_KEY_PLACEHOLDER}"

echo -e "\n-----------------------------------------"
echo "请前往 https://ollama.com/settings/keys 生成您的 API 密钥。"
echo "此密钥将直接写入 $BACKEND_FILE 文件。"
echo "-----------------------------------------"
echo -e "\n请复制您的 API 密钥，然后粘贴到下方："
echo -n "API 密钥 (Bearer Token): "
read -r USER_API_KEY

if [ -z "$USER_API_KEY" ]; then
    echo "错误：API 密钥不能为空。安装中止。"
    exit 1
fi

echo -e "\n-> 正在将密钥写入 $BACKEND_FILE 文件..."

if [ ! -f "$BACKEND_FILE" ]; then
    echo "错误：未找到后端文件 $BACKEND_FILE。请确保文件存在！"
    exit 1
fi

sed -i "s|'${PLACEHOLDER}'|'${USER_API_KEY}'|g" "$BACKEND_FILE"

if grep -q "$USER_API_KEY" "$BACKEND_FILE"; then
    echo "✅ 密钥已成功写入 $BACKEND_FILE。"
else
    echo "⚠️ 警告：自动写入密钥失败！请手动检查并修改 $BACKEND_FILE 文件中的 '$PLACEHOLDER' 占位符。"
    echo "您输入的密钥是：$USER_API_KEY"
    exit 1
fi

echo -e "\n========================================="
echo "🎉 安装脚本执行完毕！请使用包管理器手动安装 nodejs 和 jq 。"
echo "API 密钥已配置完成，文件 $BACKEND_FILE 已更新。"
echo "运行 ./on 启动服务，运行 ./off 停止服务。"
echo "========================================="
